<script setup lang="ts">
import type { VbenFormProps } from '@vben/common-ui';

import type { VxeGridProps } from '#/adapter/vxe-table';
import type { PageQuery } from '#/api/common';
import type { Device } from '#/api/iot/device/manage/model';

import { nextTick, onMounted } from 'vue';

import { useAccess } from '@vben/access';
import { Page, useVbenDrawer, useVbenModal } from '@vben/common-ui';
import { eachTree, getVxePopupContainer } from '@vben/utils';

import { useDebounceFn } from '@vueuse/core';
import { Popconfirm, Space } from 'ant-design-vue';

import { useVbenVxeGrid } from '#/adapter/vxe-table';
import { deviceRemove, devicetList } from '#/api/iot/device/manage/index';
import { getProductList } from '#/api/iot/product';
import { tenantList } from '#/api/system/tenant';
import { router } from '#/router';

import { columns, querySchema } from './data';
import deptDrawer from './dept-drawer.vue';
import deviceImportModal from './device-import-modal.vue';

const { hasAccessByRoles } = useAccess();

const formOptions: VbenFormProps = {
  commonConfig: {
    labelWidth: 80,
    componentProps: {
      allowClear: true,
    },
  },
  schema: querySchema(),
  wrapperClass: 'grid-cols-1 md:grid-cols-2 lg:grid-cols-3 xl:grid-cols-4',
};

const gridOptions: VxeGridProps = {
  columns,
  height: 'auto',
  keepSource: true,
  pagerConfig: {},
  proxyConfig: {
    ajax: {
      query: async ({ page }, formValues = {}) => {
        const resp = await devicetList({
          pageNum: page.currentPage,
          pageSize: page.pageSize,
          ...formValues,
        });
        return resp;
      },
      // 默认请求接口后展开全部 不需要可以删除这段
      querySuccess: () => {
        // 默认展开 需要加上标记

        eachTree(tableApi.grid.getData(), (item) => (item.expand = true));
        nextTick(() => {
          setExpandOrCollapse(true);
        });
      },
    },
  },
  /**
   * 虚拟滚动  默认关闭
   */
  scrollY: {
    enabled: false,
    gt: 0,
  },
  toolbarConfig: {
    custom: false,
  },
  rowConfig: {
    keyField: 'deptId',
  },
  treeConfig: {
    parentField: 'parentId',
    rowField: 'deptId',
    transform: true,
  },
  id: 'iot-device-index',
};

const [BasicTable, tableApi] = useVbenVxeGrid({
  formOptions,
  gridOptions,
  gridEvents: {
    cellDblclick: (e) => {
      const { row = {} } = e;
      if (!row?.children) {
        return;
      }
      const isExpanded = row?.expand;
      tableApi.grid.setTreeExpand(row, !isExpanded);
      row.expand = !isExpanded;
    },
    // 需要监听使用箭头展开的情况 否则展开/折叠的数据不一致
    toggleTreeExpand: (e) => {
      const { row = {}, expanded } = e;
      row.expand = expanded;
    },
  },
});

const initFormData = async (val?: string) => {
  const params: PageQuery = {};
  if (val) {
    params.productName = val;
  }
  const res = await getProductList(params);
  const options = res.rows || [];

  // 产品信息
  tableApi.formApi.updateSchema([
    {
      componentProps: {
        fieldNames: { label: 'productName', value: 'id' },
        options,
        showSearch: true,
        filterOption: false,
        onSearch: useDebounceFn(async (val: string) => {
          initFormData(val);
        }, 300),
      },
      fieldName: 'productId',
    },
  ]);
};

const initTenantData = async (val?: string) => {
  const params: PageQuery = {};
  if (val) {
    params.companyName = val;
  }
  const res = await tenantList(params);
  const options = res.rows || [];

  // 产品信息
  tableApi.formApi.updateSchema([
    {
      componentProps: {
        fieldNames: { label: 'companyName', value: 'id' },
        options,
        showSearch: true,
        filterOption: false,
        onSearch: useDebounceFn(async (val: string) => {
          initTenantData(val);
        }, 300),
      },
      fieldName: 'tenantId',
    },
  ]);
};

onMounted(() => {
  initFormData();
  if (hasAccessByRoles(['superadmin'])) {
    initTenantData();
  }
});

const [DeptDrawer, drawerApi] = useVbenDrawer({
  connectedComponent: deptDrawer,
});

function handleAdd() {
  drawerApi.setData({ update: false });
  drawerApi.open();
}

async function handleEdit(record: Device) {
  drawerApi.setData({ id: record.id, update: true });
  drawerApi.open();
}

async function handleDelete(row: Device) {
  await deviceRemove(row.id);
  await tableApi.query();
}

/**
 * 全部展开/折叠
 * @param expand 是否展开
 */
function setExpandOrCollapse(expand: boolean) {
  eachTree(tableApi.grid.getData(), (item) => (item.expand = expand));
  tableApi.grid?.setAllTreeExpand(expand);
}

/**
 * 导入
 */
const [DeviceImpotModal] = useVbenModal({
  connectedComponent: deviceImportModal,
});

const handleInfo = (row: Device) => {
  router.push(`/device/detail/${row.id}`);
};
</script>

<template>
  <Page :auto-content-height="true">
    <BasicTable table-title="设备列表">
      <template #toolbar-tools>
        <Space>
          <!-- <a-button
            v-access:code="['iot:device:export']"
            @click="handleDownloadExcel"
          >
            {{ $t('pages.common.export') }}
          </a-button>
          <a-button v-access:code="['iot:device:import']" @click="handleImport">
            {{ $t('pages.common.import') }}
          </a-button> -->
          <a-button
            type="primary"
            v-access:code="['iot:device:add']"
            @click="handleAdd"
          >
            {{ $t('pages.common.add') }}
          </a-button>
        </Space>
      </template>
      <template #action="{ row }">
        <Space>
          <ghost-button
            v-access:code="['iot:device:query']"
            @click="handleInfo(row)"
          >
            {{ $t('pages.common.info') }}
          </ghost-button>
          <ghost-button
            v-access:code="['iot:device:edit']"
            @click="handleEdit(row)"
          >
            {{ $t('pages.common.edit') }}
          </ghost-button>
          <Popconfirm
            :get-popup-container="getVxePopupContainer"
            placement="left"
            title="确认删除？"
            @confirm="handleDelete(row)"
          >
            <ghost-button
              danger
              v-access:code="['iot:device:remove']"
              @click.stop=""
            >
              {{ $t('pages.common.delete') }}
            </ghost-button>
          </Popconfirm>
        </Space>
      </template>
    </BasicTable>
    <DeptDrawer @reload="tableApi.query()" />
    <DeviceImpotModal @reload="tableApi.query()" />
  </Page>
</template>
